from easydict import EasyDict as edict

__C = edict()
# Consumers can get config by: from config import cfg

cfg = __C

# YOLO options
__C.yolo = edict()

# Set the class name
__C.yolo.class_path = "./data/classes/ifly_classes.txt"
__C.yolo.anchors_path = "./data/anchors/yolo_anchors.txt"
__C.yolo.moving_ave_decay = 0.9995
__C.yolo.strides = [8, 16, 32]
__C.yolo.maxAnchorNum_per_scale = 150 # 生成标签框时，每一种比例下最多生成的锚框
__C.yolo.iou_loss_thresh = 0.5
__C.yolo.original_weight = "checkpoint/yolov4.ckpt"

# Train options
__C.train = edict()

__C.train.train_path = "./data/datasets/ifly_train.txt"
__C.train.batch_size = 1
__C.train.epochs = 50
__C.train.input_size = 416
__C.train.learn_rt = 1e-4

# TEST options
__C.test = edict()

__C.test.test_path = "data/dataset/ifly_test.txt"
__C.test.score_threshold = 0.3
__C.test.iou_threshold = 0.5

